package com.linyaonan.leetcode.medium._535;

import java.util.HashMap;
import java.util.Random;

/**
 * TinyURL是一种URL简化服务， 比如：当你输入一个URL https://leetcode.com/problems/design-tinyurl 时，它将返回一个简化的URL http://tinyurl.com/4e9iAk.
 *
 * 要求：设计一个 TinyURL 的加密 encode 和解密 decode 的方法。你的加密和解密算法如何设计和运作是没有限制的，你只需要保证一个URL可以被加密成一个TinyURL，并且这个TinyURL可以用解密方法恢复成原本的URL。
 *
 *
 * @author: Lin
 * @date: 2020/1/14
 */
public class Tinyurl {
    HashMap<String, String> map = new HashMap<>();
    String hex = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
    public String encode(String longUrl) {
        String roundStr = getRoundStr();
        map.put(roundStr, longUrl);
        return "http://tinyurl.com/" + roundStr;
    }

    public String decode(String shortUrl) {
        String[] s = shortUrl.split("http://tinyurl.com/", -1);
        return map.get(s[1]);
    }
    // 生成随机子串
    private String getRoundStr() {
        boolean isGet = false;
        String str = null;
        while (!isGet) {
            StringBuilder sb = new StringBuilder();
            for(int i = 0; i < 6; i++) {
                int i1 = Math.abs(new Random().nextInt());
                sb.append(hex.charAt(i1 % 61));
            }
            str = sb.toString();
            if (!map.containsKey(str)) {
                isGet = true;
            }
        }
        return str;
    }
}
